f, ax = plt.subplots(figsize=(4,4))
for sp in data['species'].unique():
    tmp_data = data.query('species == @sp')
    ax.scatter(x=tmp_data['flipper_length_mm'], y=tmp_data['body_mass_g'], label=sp)
    
ax.set_title('Penguin measurements');
ax.set_ylabel('Body Mass [g]');
ax.set_xlabel('Flipper Length [mm]');
l = ax.legend(bbox_to_anchor=(0.0, 1.0), loc='upper left');
l.set_frame_on(False)